package com.gnxk.datasource.config;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

//数据源配置类
@Configuration
public class MyDataSource{

    //这是自定义的数据源
    @Bean(name = "sdeDataSource")
    @ConfigurationProperties("spring.datasource.my")
    public DataSource getMyDataSource(){
        DataSource dataSource = DataSourceBuilder.create().build();
        return dataSource;
    }

    //这个是默认的数据源，会自动配置，因为我们这里要使用jdbctemplate,为了避免循环依赖，还是
    //显示的配置了一下，并且在启动类上增加了exclude={DataSourceAutoConfiguration.class}
    //使自动数据源配置失效
    @Bean(name = "dataSource")
    @ConfigurationProperties("spring.datasource")
    public DataSource getDataSource(){
        DataSource dataSource = DataSourceBuilder.create().build();
        return dataSource;
    }

    @Primary //当相同类型的实现类存在时，选择该注解标记的类
    @Bean("dynamicDS")
    public DataSource dataSource(){
        DynamicDataSource dds=new DynamicDataSource();
        //默认数据源，mysql
        dds.setDefaultTargetDataSource(getDataSource());
        Map<Object,Object> dsMap=new HashMap<>();
        dsMap.put("dataSource",getDataSource());
        dsMap.put("sdeDataSource",getMyDataSource());
        dds.setTargetDataSources(dsMap);
        return dds;
    }

    //事务管理器DataSourceTransactionManager构造参数需要DataSource
    //这里可以看到我们给的是dynamicDS这个bean
    @Bean
    public PlatformTransactionManager transactionManager(){
        return new DataSourceTransactionManager(dataSource());
    }

    //这里的JdbcTemplate构造参数同样需要一个DataSource,为了实现数据源切换查询，
    //这里使用的也是dynamicDS这个bean
    @Bean(name = "jdbcTemplate")
    public JdbcTemplate getJdbc(){
        return new JdbcTemplate(dataSource());
    }
}
